from pyecharts.charts import Bar, Timeline
from pyecharts.globals import ThemeType
from pyecharts.options import LabelOpts, TitleOpts

# 读取数据
f = open("E:/storage/study/Python/python-learn/day01/02-python-learn/10_可视化案例/素材/1960-2019全球GDP数据.csv","r",encoding="GB2312")
data_lines = f.readlines()
# 关闭文件
f.close()
# 删除第一条数据
data_lines.pop(0)
# 将数据转换为字典存储，格式为：
# {年份：[国家，gdp],[国家, gdp], ......], 年份：[国家，gdp],[国家, gdp], ......], .......}
# {年份：[美国，123],[中国, 321], ......], 年份：[美国，123],[中国, 321], ......], .......}
# 先定义一个字典对象
data_dict = {}
for line in data_lines:
    year = int(line.split(",")[0]) # 年份
    country = line.split(",")[1] # 国家
    gdp = float(line.split(",")[2]) # gdp数据
    # 如何判断字典里面有没有指定的key呢？
    try:
        data_dict[year].append([country,gdp])
    except KeyError:
        data_dict[year] = []
        data_dict[year].append([country,gdp])
# print(data_dict[1960])
timeline = Timeline(
    {"theme": ThemeType.LIGHT}
)
# 排序年份
sorted_year_list = sorted(data_dict.keys())
for year in sorted_year_list:
    data_dict[year].sort(key=lambda  element: element[1], reverse=True)
    # 取出本年分前8名的国家
    year_data = data_dict[year][0:8]
    x_data = []
    y_data = []
    for country_gdp in year_data:
        print(country_gdp)
        x_data.append(country_gdp[0]) # x轴添加国家
        y_data.append(country_gdp[1] / 100000000) # y轴添加gdp数据

    # 构建柱状图
    bar = Bar()
    bar.add_xaxis(x_data)
    bar.add_yaxis("GDP（亿）", y_data, label_opts=LabelOpts(position="right"))
    # 反转x轴和y轴
    bar.reversal_axis()
    # 设置每一年的图标的标题
    bar.set_global_opts(
        title_opts=TitleOpts(is_show=True,title=f"{year}年全球前8GDP数据")
    )

    timeline.add(bar, str(year))

timeline.add_schema(
    play_interval = 200,
    is_timeline_show = True,
    is_auto_play = True,
    is_loop_play = True
)
timeline.render("全球经济.html")